VR-based cognitive training program